cd ../..

gpus=$1

python -m trainer.BaselineTrainer fit \
    --config configs/experiment/mamba.yaml \
    --config configs/model/CMamba.yaml \
    --config configs/dataset/desed.yaml \
    --config configs/trainer/default.yaml \
    --model.exp_name desed/baseline/cmamba_adamW \
    --trainer.devices=${gpus} 